from tool import get_completion

# 从代码生成角度,先对用户的语言进行整理，按步骤整理和梳理
text_1 = f"""
首先，我有6个参数，<批次类型>，<猪场类型>，<P1断奶7天发情配种率>，<P2+断奶7天发情配种率>和<引种补充配种头数>，需要计算<本批次断奶7天配种头数>；
一开始我需要计算<本批次断奶7天配种头数>，它的计算方法是<引种补充配种头数>**0.5 + 1/2 * <引种补充配种头数>；
然后我需要判断<批次类型>是<存栏批次>还是<未来批次>,如果是存栏批次，那么<本批次断奶7天配种头数>=0；如果是<未来批次>，我需要进一步判断<猪场类型>；
如果<猪场类型>是<P1(一胎)场>，那么<本批次断奶7天配种头数>=<上个批次末存栏头数> * <P1断奶7天发情配种率>;
否则，那么<本批次断奶7天配种头数>=<上个批次末存栏头数> * <P2+断奶7天发情配种率>。
"""

prompt1 = f"""
您将获得由三个引号括起来的文本。\
如果它包含一系列的指令，则需要按照以下格式重新编写这些指令：
输入参数 - ...
返回参数 - ...
第一步 - ...
第二步 - …
…
第N步 - …
如果文本中不包含一系列的指令，则直接写“未提供步骤”。"
\"\"\"{text_1}\"\"\"
"""



if __name__ == '__main__':
    response = get_completion(prompt1)
    print("Text 1 的总结:")
    print(response)

    prompt_code = f"""
        您将获得由三个引号括起来的文本，并根据这段文本作为需求，python3写一个函数来实现这个需求\
        并以 JSON 格式提供，其中包含以下键:language、code。
        \"\"\"{response}\"\"\"
        """

    response_code = get_completion(prompt_code)
    print("prompt_code 的代码:")
    print(response_code)